import os

# 检查 HADOOP_HOME 环境变量
hadoop_home = os.environ.get("HADOOP_HOME")
if hadoop_home:
    print(f"HADOOP_HOME: {hadoop_home}")
else:
    print("未设置 HADOOP_HOME 环境变量")

# 检查 PATH 环境变量中是否包含 hadoop 命令
path_dirs = os.environ.get("PATH", "").split(os.pathsep)
hadoop_bin = next((d for d in path_dirs if "hadoop" in d.lower() and os.path.exists(os.path.join(d, "hadoop"))), None)
if hadoop_bin:
    print(f"找到 hadoop 命令在: {hadoop_bin}")
    # 通常 hadoop/bin 目录的父级即为 Hadoop 根目录
    hadoop_root = os.path.dirname(hadoop_bin)
    print(f"推测 Hadoop 根目录: {hadoop_root}")
else:
    print("未在 PATH 中找到 hadoop 命令")